package view_server

import (
	"bigdevops/src/common"
	"bigdevops/src/config"
	"bigdevops/src/models"
	"encoding/json"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/go-playground/validator/v10"
	"go.uber.org/zap"
	"strconv"
)

func createJobTask(c *gin.Context) {

	sc := c.MustGet(common.GIN_CTX_CONFIG_CONFIG).(*config.ServerConfig)
	var reqObj models.JobTask
	err := c.ShouldBindJSON(&reqObj)
	if err != nil {
		sc.Logger.Error("解析新增任务执行请求失败", zap.Any("任务执行", reqObj), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	// 在这里校验字段，是否必填，范围是否正确
	err = validate.Struct(reqObj)
	if err != nil {

		// 这里为什么要判断错误是否是 ValidationErrors
		if errors, ok := err.(validator.ValidationErrors); ok {
			common.ReqBadFailWithWithDetailed(

				gin.H{
					"翻译前": err.Error(),
					"翻译后": errors.Translate(trans),
				},
				"请求出错",
				c,
			)
			return
		}
		common.ReqBadFailWithMessage(err.Error(), c)
		return

	}

	userName := c.MustGet(common.GIN_CTX_JWT_USER_NAME).(string)
	dbUser, err := models.GetUserByUserName(userName)
	if err != nil {
		sc.Logger.Error("通过token解析到的userName去数据库中找User失败",
			zap.Error(err),
		)
		common.ReqBadFailWithMessage(fmt.Sprintf("通过token解析到的userName去数据库中找User失败:%v", err.Error()), c)
		return
	}

	reqObj.UserID = dbUser.ID

	// treeNode是必须的，直接根据 HostIds 那么直接从node上获取
	var hostIds []string
	err = json.Unmarshal([]byte(reqObj.HostsIdsRaw), &hostIds)
	if err != nil {
		errMsg := fmt.Sprintf("通过token解析到的hostIds去数据库中找User失败:%v", err.Error())
		sc.Logger.Error(errMsg,
			zap.Error(err),
		)
		common.ReqBadFailWithMessage(errMsg, c)
		return
	}
	if len(hostIds) == 0 {
		errMsg := fmt.Sprintf("通过HostsIdsRaw解析到的hostIds为空")
		sc.Logger.Error(errMsg,
			zap.Error(err),
		)
		common.ReqBadFailWithMessage(errMsg, c)
		return
	}

	hostIdsInt := []int{}
	for _, idS := range hostIds {
		intS, _ := strconv.Atoi(idS)
		hostIdsInt = append(hostIdsInt, intS)
	}
	hosts, err := models.GetResourceEcsByIdsWithLimitOffset(hostIdsInt, 10000, 0)
	if err != nil {
		errMsg := fmt.Sprintf("通过hostIds找机器错误")
		sc.Logger.Error(errMsg,
			zap.Error(err),
		)
		common.ReqBadFailWithMessage(errMsg, c)
		return
	}
	hostIps := []string{}
	for _, host := range hosts {
		hostIps = append(hostIps, host.IpAddr)
	}
	hostIpsRaw, _ := json.Marshal(hostIps)
	reqObj.TotalNum = len(hosts)
	reqObj.HostsRaw = string(hostIpsRaw)
	reqObj.Status = common.JOB_STATUS_PENING
	// 下面级联创建
	err = reqObj.CreateOne()
	if err != nil {
		sc.Logger.Error("新增任务执行数据库失败", zap.Any("任务执行", reqObj), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	common.OkWithMessage("创建成功", c)

}

func getJobTaskList(c *gin.Context) {

	sc := c.MustGet(common.GIN_CTX_CONFIG_CONFIG).(*config.ServerConfig)

	currentPage, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "100"))

	offset := 0
	limit := 0
	limit = pageSize
	if currentPage > 1 {
		offset = (currentPage - 1) * limit
	}
	// 这里只能新增 不能去掉，因为是 多种资源共用的query
	searchUserID := c.DefaultQuery("UserID", "")
	searchUserIDInt, _ := strconv.Atoi(searchUserID)
	searchName := c.DefaultQuery("name", "")

	// 	数据库中拿到所有的menu列表

	objs, err := models.GetJobTaskAll()
	if err != nil {
		sc.Logger.Error("去数据库中拿所有的任务执行错误",
			zap.Error(err),
		)
		common.ReqBadFailWithMessage(fmt.Sprintf("去数据库中拿所有的任务执行错误:%v", err.Error()), c)
		return
	}

	allIds := []int{}
	//// 遍历 role 准备menuIds 列表
	for _, obj := range objs {
		obj := obj

		// 在这里处理查询
		if searchName != "" && obj.Title != searchName {
			continue
		}

		// 在这里处理查询
		if searchUserID != "" && int(obj.UserID) != searchUserIDInt {
			continue
		}
		allIds = append(allIds, int(obj.ID))

		//role.MenuIds = menuIds
	}

	if len(allIds) == 0 {
		common.OkWithDetailed(allIds, "ok", c)
		return
	}
	objs, err = models.GetJobTaskByIdsWithLimitOffset(allIds, limit, offset)
	if err != nil {
		sc.Logger.Error("limit-offset去数据库中拿所有的任务错误",
			zap.Error(err),
		)
		common.ReqBadFailWithMessage(fmt.Sprintf("limit-offset去数据库中拿所有的脚本模板错误:%v", err.Error()), c)
		return
	}
	for _, obj := range objs {
		obj.FillFrontAllData()

	}

	resp := &ResponseResourceCommon{
		//Total: models.GetJobTaskCount(), 因为是带了查询条件的 所以 count 不能是all 而是 过滤后的total

		Total: len(allIds),
		Items: objs,
	}

	common.OkWithDetailed(resp, "ok", c)
}

func getJobTaskOne(c *gin.Context) {
	sc := c.MustGet(common.GIN_CTX_CONFIG_CONFIG).(*config.ServerConfig)
	// 校验一下 menu字段
	id := c.Param("id")

	// 先 去db中根据id找到这个user
	intVar, _ := strconv.Atoi(id)
	dbObj, err := models.GetJobTaskById(intVar)
	if err != nil {
		sc.Logger.Error("根据id找任务实例错误", zap.Any("任务实例", id), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}
	//isRelatedWithMe := false

	//userName := c.MustGet(common.GIN_CTX_JWT_USER_NAME).(string)
	//dbUser, err := models.GetUserByUserName(userName)
	//if err != nil {
	//	sc.Logger.Error("通过token解析到的userName去数据库中找User失败",
	//		zap.Error(err),
	//	)
	//	common.ReqBadFailWithMessage(fmt.Sprintf("通过token解析到的userName去数据库中找User失败:%v", err.Error()), c)
	//	return
	//}

	dbObj.FillFrontAllData()

	common.OkWithData(dbObj, c)
}

func deleteJobTask(c *gin.Context) {
	sc := c.MustGet(common.GIN_CTX_CONFIG_CONFIG).(*config.ServerConfig)
	// 校验一下 menu字段
	id := c.Param("id")
	sc.Logger.Info("删除任务执行", zap.Any("id", id))

	// 先 去db中根据id找到这个对象
	intVar, _ := strconv.Atoi(id)
	dbObj, err := models.GetJobTaskById(intVar)
	if err != nil {
		sc.Logger.Error("根据id找任务执行错误", zap.Any("任务执行", id), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	//  做个保护，查询一下template 中有没有用到 这个id的
	dbTemplate, _ := models.GetJobTaskById(intVar)
	if dbTemplate != nil {
		sc.Logger.Error("根据JobTaskid找到了模板，禁止删除ormDesign", zap.Any("任务执行", id), zap.Error(err))
		common.FailWithMessage("根据JobTaskid找到了模板，禁止删除ormDesign ", c)
		return
	}

	err = dbObj.DeleteOne()
	if err != nil {
		sc.Logger.Error("根据id删除任务执行错误", zap.Any("任务执行", id), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}
	common.OkWithMessage("删除成功", c)
}

func updateJobTask(c *gin.Context) {
	sc := c.MustGet(common.GIN_CTX_CONFIG_CONFIG).(*config.ServerConfig)
	var reqObj models.JobTask
	err := c.ShouldBindJSON(&reqObj)
	if err != nil {
		sc.Logger.Error("解析新增任务执行请求失败", zap.Any("任务执行", reqObj), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	// 在这里校验字段，是否必填，范围是否正确
	err = validate.Struct(reqObj)
	if err != nil {

		// 这里为什么要判断错误是否是 ValidationErrors
		if errors, ok := err.(validator.ValidationErrors); ok {
			common.ReqBadFailWithWithDetailed(

				gin.H{
					"翻译前": err.Error(),
					"翻译后": errors.Translate(trans),
				},
				"请求出错",
				c,
			)
			return
		}
		common.ReqBadFailWithMessage(err.Error(), c)
		return

	}

	_, err = models.GetJobTaskById(int(reqObj.ID))
	if err != nil {
		sc.Logger.Error("根据id找任务执行错误", zap.Any("任务执行", reqObj), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	err = reqObj.UpdateOne()

	if err != nil {
		sc.Logger.Error("更新任务执行本身错误", zap.Any("任务执行", reqObj), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	common.OkWithMessage("更新成功", c)
}

func actionJobTaskOne(c *gin.Context) {
	sc := c.MustGet(common.GIN_CTX_CONFIG_CONFIG).(*config.ServerConfig)
	// 校验一下 menu字段
	id := c.Param("id")

	// 先 去db中根据id找到这个对象
	intVar, _ := strconv.Atoi(id)
	dbObj, err := models.GetJobTaskById(intVar)
	if err != nil {
		sc.Logger.Error("根据id找任务执行错误", zap.Any("任务执行", id), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	//nextStatus := c.Query("nextStatus")
	action := c.Query("action")
	// 应该根据action去找下一个状态

	// 判断下一个状态的是否在 map中

	nextStatus, exist := common.JOB_ACTION_NEXT_STATUS_MAP[action]
	if !exist {
		sc.Logger.Error("传入的动作错误", zap.Any("任务执行", id), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	if action == common.AGENT_TASK_ACTION_KILL {
		dbObj.Action = common.AGENT_TASK_ACTION_KILL
	}

	sc.Logger.Info("任务动作",
		zap.Any("id", id),
		zap.Any("action", action),
		zap.Any("nextStatus", nextStatus),
	)
	userName := c.MustGet(common.GIN_CTX_JWT_USER_NAME).(string)
	// 记录任务流
	dbObj.UpdateFlowData(action, userName)
	dbObj.Status = nextStatus
	err = dbObj.UpdateOne()
	if err != nil {
		sc.Logger.Error("更新任务执行错误", zap.Any("任务执行", id), zap.Error(err))
		common.FailWithMessage(err.Error(), c)
		return
	}

	common.OkWithMessage("更新成功", c)
}
